'''
Author: Diana Tang
Date: 2025-04-04 22:41:07
LastEditors: Diana Tang
Description: some description
FilePath: /pub-scholar-assist/setup.py
'''
from setuptools import setup, find_packages

setup(
    name="deepseek-kb",
    version="0.1.0",
    packages=find_packages(),
    install_requires=[
        "transformers>=4.34.0",
        "torch>=2.0.0",
        "faiss-cpu>=1.7.4",
        "sentence-transformers>=2.2.2",
        "langchain>=0.0.312",
        "langchain-community>=0.0.10",
        "pyyaml>=6.0",
        "numpy>=1.24.0",
        "pypdf>=3.15.1",
        "docx2txt>=0.8",
        "unstructured>=0.10.0",
        "requests>=2.31.0",
        "numpy>=1.20.0",
        "pandas>=1.3.0",
        "pyyaml>=6.0",
        "tqdm>=4.62.0",
        "faiss-cpu>=1.7.0",
        "sentence-transformers>=2.2.0",
        "PyMuPDF>=1.20.0",  # Added for PDF processing with fitz
        "langchain>=0.0.200",  # Kept for compatibility
        "loguru>=0.6.0",
    ],
    author="Your Name",
    author_email="your.email@example.com",
    description="知识库系统，基于本地DeepSeek模型",
    keywords="deepseek, knowledge-base, RAG",
    python_requires=">=3.8",
)